package com.agriculture.controller;

import com.agriculture.common.R;
import com.agriculture.entity.*;
import com.agriculture.service.SellerService;
import com.agriculture.utils.JwtUtils;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import io.jsonwebtoken.Claims;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.util.DigestUtils;
import org.springframework.web.bind.annotation.*;

import javax.servlet.http.HttpServletRequest;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

@Slf4j
@CrossOrigin
@RestController
@RequestMapping("/seller")
public class SellerController {

    @Autowired
    private SellerService sellerService;

    /**
     * 商家登录
     * @param seller
     * @return
     */
    @PostMapping("/login")
    public R<String> login( @RequestBody Seller seller){
        //1、将页面提交的密码password进行md5加密处理
        String password = seller.getPassword();
        password = DigestUtils.md5DigestAsHex(password.getBytes());

        //2、根据页面提交的用户名username查询数据库
        LambdaQueryWrapper<Seller> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.eq(Seller::getName,seller.getName());
        Seller s = sellerService.getOne(queryWrapper);

        //3、如果没有查询到则返回登录失败结果
        if(s == null){
            return R.error("登录失败");
        }

        //4、密码比对，如果不一致则返回登录失败结果
        if(!s.getPassword().equals(password)){
            return R.error("登录失败");
        }

        // 6、生成令牌，发放令牌
        Map<String, Object> claims = new HashMap<>();
        claims.put("id", s.getId());
        claims.put("name", s.getName());
        claims.put("phone", s.getPhone());
        claims.put("sex", s.getSex());
        claims.put("id_number", s.getIdNumber());
        claims.put("avatar", s.getAvatar());
        claims.put("salary", s.getSalary());
        claims.put("superior_id", s.getSuperiorId());
        claims.put("area_code", s.getAreaCode());
        claims.put("level", s.getLevel());
        claims.put("is_deleted", s.getIsDeleted());
        String jwt = JwtUtils.generateJwt(claims);

        return R.success(jwt);
    }

    /**
     * 根据 token 返回用户数据
     * @param token
     * @return
     */
    @GetMapping("/info/{token}")
    public R getInfoBtToken(@PathVariable String token){
        Claims claims = JwtUtils.parseJWT(token);
        return R.success(claims);
    }

    /**
     * 商家注册
     * @param seller
     * @return
     */
    @PostMapping("/register")
    public R<String> register(@RequestBody Seller seller){
        if(seller.getName() == "" || seller.getPassword() == ""){
            return R.error("用户名或密码不能为空");
        }

        //1、将页面提交的密码password进行md5加密处理
        seller.setPassword(DigestUtils.md5DigestAsHex("123456".getBytes()));

        //2、根据页面提交的用户名username查询数据库
        LambdaQueryWrapper<Seller> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.eq(Seller::getName,seller.getName());
        Seller s = sellerService.getOne(queryWrapper);

        //3、如果没有查询到则返回登录失败结果
        if(s == null){
            // 能注册，将数据写进数据库
            sellerService.save(seller);
            return R.success("注册成功");
        }else{
            // 不能注册
            return R.error("用户名重复");
        }
    }

    /**
     * 商家退出登录
     * @param request
     * @return
     */
    @PostMapping("/logout")
    public R<String> logout(HttpServletRequest request){
        //清理Session中保存的当前登录员工的id
        request.getSession().removeAttribute("seller");
        return R.success("退出成功");
    }

    /**
     * 返回指定级别的下一级
     * @param
     * @return
     */
    @GetMapping("list")
    public R<List<Seller>> list(HttpServletRequest request){
        Long id = (Long) request.getSession().getAttribute("seller");
        Seller s = sellerService.getById(id);

        //条件构造器
        LambdaQueryWrapper<Seller> queryWrapper = new LambdaQueryWrapper<>();

        if(s.getLevel() == 0){  // 超级管理员
            queryWrapper.isNull(Seller::getAreaCode);
        }else if(s.getLevel() == 1){  // 一级经销商
            queryWrapper.eq(s.getAreaCode() != null, Seller::getAreaCode, s.getAreaCode());
        }else if(s.getLevel() == 2){  // 二级经销商
            queryWrapper.eq(s.getAreaCode() != null,Seller::getAreaCode, s.getAreaCode());
        }

        return R.success(sellerService.list(queryWrapper));
    }

    /**
     * 返回所有商家信息
     * @return
     */
    @GetMapping("/getAll")
    public R<List<Seller>> getAllSeller(){
        return R.success(sellerService.list());
    }


    /**
     * 根据 id 查找 seller 信息
     * @param id
     * @return
     */
    @GetMapping("/getSellerById/{id}")
    public R getSellerById(@PathVariable Long id){
        Seller seller = sellerService.getById(id);
        if(seller == null){
            return R.error("id 为" + id + "的 seller 信息找不到！");
        }else{
            return R.success(seller);
        }
    }

    /**
     * 分页查询 seller 信息
     * @param page
     * @param pageSize
     * @return
     */
    @GetMapping("/page")
    public R<Page> page(int page, int pageSize){

        LambdaQueryWrapper<Seller> sellerWrapper = new LambdaQueryWrapper<>();

        //分页构造器
        Page<Seller> pageInfo = new Page<>(page,pageSize);

        //添加排序条件，根据Level进行排序
        sellerWrapper.orderByAsc(Seller::getLevel);

        //分页查询
        Page<Seller> sellerPage = sellerService.page(pageInfo, sellerWrapper);
        return R.success(sellerPage);
    }


    /**
     * 新增商家信息
     * @param seller
     * @return
     */
    @PostMapping("/addSeller")
    public R<String> addSeller(@RequestBody Seller seller){
        String password = "123456";
        seller.setPassword(DigestUtils.md5DigestAsHex(password.getBytes()));
        sellerService.save(seller);
        return R.success("新增成功！");
    }
}
